cmake_minimum_required(VERSION 2.8)
project(classify)
list(APPEND CMAKE_MODULE_PATH ${CMAKE_CURRENT_SOURCE_DIR})
find_package(Caffe REQUIRED)
find_package(CUDA REQUIRED)
find_package(OpenCV REQUIRED)
include_directories(${CAFFE_INCLUDE_DIR} ${CUDA_TOOLKIT_INCLUDE} ${OPENCV_INCLUDE_DIR})
message("${CAFFE_INCLUDE_DIR},${CUDA_TOOLKIT_INCLUDE},${OPENCV_INCLUDE_DIR}")
add_executable(${PROJECT_NAME} main.cpp)
target_link_libraries(${PROJECT_NAME} ${CAFFE_LIBRARY} ${CUDA_cublas_LIBRARY} ${OPENCV_LIBRARY} glog)
